# 导入ocr_nlp_processor 
from ocr_nlp_processor import ocr_nlp_processor

# 定义prompt和表头的映射
prompt_to_header = {
    "姓名是什么？": {"header": "姓名"},
    "在哪里上的大学？": {"header": "大学"},
    "学历是什么？": {"header": "学历"},
}

# 输入文件目录
input_directory = './简历/所有文件'
# 输出文件目录
output_directory = './简历/已处理文件'
# 输出CSV文件路径
output_csv = './简历/识别结果.csv'

# 调用通用的OCR和NLP处理函数
ocr_nlp_processor(input_directory, output_directory, output_csv, prompt_to_header)
